package cn.com.leetcode.dp;

/**
 * 三步问题。有个小孩正在上楼梯，楼梯有n阶台阶，小孩一次可以上1阶、2阶或3阶。实现一种方法，计算小孩有多少种上楼梯的方式。结果可能很大，你需要对结果模1000000007。
 * <p>
 * 示例1:
 * <p>
 * 输入：n = 3
 * 输出：4
 * 说明: 有四种走法
 * 示例2:
 * <p>
 * 输入：n = 5
 * 输出：13
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/three-steps-problem-lcci
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Demo_Ms08_01 {
    public static void main(String[] args) {
        Demo_Ms08_01 demo_ms08_01 = new Demo_Ms08_01();
        int step = demo_ms08_01.waysToStep(100);
        System.out.println(step);
    }

    public int waysToStep(int n) {
        int before = 1;
        int center = 2;
        int after = 4;

        if (n == 1 || n == 2) {
            return n;
        }
        for (int i = 3; i < n; i++) {
            int temp = (before + (center + after) % 1000000007) % 1000000007;
            before = center % 1000000007;
            center = after % 1000000007;
            after = temp;
        }
        return after;
    }

}
